#ifndef AGXKIT_H
#define AGXKIT_H

#include <string>

class AGXKit {
private:
    std::string model;         // 型号
    double ai;                 // AI性能
    int cudaCores;             // CUDA核心数
    int tensorCores;           // Tensor CORE数
    int memory;                // 显存
    int storage;               // 存储

public:
    AGXKit();
    AGXKit(const std::string& model, double ai, int cudaCores,
           int tensorCores, int memory, int storage);

    // Getters
    std::string getModel() const;
    double getAi() const;
    int getCudaCores() const;
    int getTensorCores() const;
    int getMemory() const;
    int getStorage() const;

    // Setters
    void setModel(const std::string& model);
    void setAi(double ai);
    void setCudaCores(int cudaCores);
    void setTensorCores(int tensorCores);
    void setMemory(int memory);
    void setStorage(int storage);

    void display() const;
};

#endif // AGXKIT_H